import os

import pandas as pd
from openpyxl.reader.excel import load_workbook

excel_path = 'pilogstat.xlsx'
new_data = pd.DataFrame({
            'device_id': [1],
            '1': [1],
            '3': [1],
            '4': [1],
            '2': [1],
            'total': [1],
        })
if os.path.exists(excel_path):
    # 加载现有的Excel文件
    book = load_workbook(filename=excel_path)
    # 假设我们要写入的是第一个工作表，也可以根据需要选择其他工作表
    sheet = book.active
    mode = 'a'  # 追加模式
    if_sheet_exists = 'overlay'  # 已存在工作表时，追加数据
    book.close()  # 关闭工作簿以释放资源
else:
    mode = 'w'  # 写入模式
    if_sheet_exists = None  # 不需要设置，因为工作表不存在

# 写入Excel文件，如果文件不存在，ExcelWriter将创建它
with pd.ExcelWriter(excel_path, engine='openpyxl', mode=mode, if_sheet_exists=if_sheet_exists) as writer:
    # 写入DataFrame到Excel的下一个空白列
    new_data.to_excel(writer, index=False, sheet_name='153pi')

print("数据已成功写入Excel文件。")